# -*- coding:utf-8 -*-
import BasePage


class Baidu_Page(BasePage.BasePage):
    search_box = "id>>>kw"  # 通过id定位搜索框
    search_button = "id>>>su"  # 通过id定位"百度一下"button

    # 输入搜索内容到搜索框
    def input_search_text(self, text):
        input_box = self.driver.get_element(self.search_box)
        input_box.send_keys(text)

    # 点击搜索按钮
    def click_search_button(self):
        self.driver.get_element(self.search_button).click()